package com.sl.mongodb.service.impl;


import com.sl.mongodb.entity.Person;
import com.sl.mongodb.service.PersonService;
import org.bson.types.ObjectId;
import org.springframework.data.domain.PageRequest;
import org.springframework.data.mongodb.core.MongoTemplate;
import org.springframework.data.mongodb.core.query.Criteria;
import org.springframework.data.mongodb.core.query.Query;
import org.springframework.data.mongodb.core.query.Update;
import org.springframework.stereotype.Service;

import javax.annotation.Resource;
import java.util.List;

@Service
public class PersonServiceImpl implements PersonService {

    @Resource
    private MongoTemplate mongoTemplate;

    /**
     * 新增数据
     *
     * @param person 数据
     */
    @Override
    public void savePerson(Person person) {
        this.mongoTemplate.save(person);
    }

    /**
     * 更新数据
     *
     * @param person 数据
     */
    @Override
    public void update(Person person) {
        //条件
        Query query = Query.query(Criteria.where("id").is(person.getId()));

        //更新的数据
        Update update = Update.update("age", person.getAge())
                .set("name", person.getName())
                .set("location", person.getLocation())
                .set("address", person.getAddress());

        //更新数据
        this.mongoTemplate.updateFirst(query, update, Person.class);
    }

    /**
     * 根据名字查询用户列表
     *
     * @param name 用户名字
     * @return
     */
    @Override
    public List<Person> queryPersonListByName(String name) {
        Query query = Query.query(Criteria.where("name").is(name)); //构造查询条件
        return this.mongoTemplate.find(query, Person.class);
    }

    /**
     * 分页查询用户列表
     *
     * @param page     页数
     * @param pageSize 页面大小
     * @return
     */
    @Override
    public List<Person> queryPersonPageList(int page, int pageSize) {
        // 从0开始的 因此要减1
        PageRequest pageRequest = PageRequest.of(page - 1, pageSize);
        Query query = new Query().with(pageRequest);
        return this.mongoTemplate.find(query, Person.class);
    }

    /**
     * 根据id删除用户
     *
     * @param id 主键
     */
    @Override
    public void deleteById(String id) {
        // 这个id用自己的id
        Query query = Query.query(Criteria.where("_id").is(new ObjectId(id)));
        this.mongoTemplate.remove(query, Person.class);
    }

    /**
     * 批量保存用户
     *
     * @param personList 用户集合
     */
    @Override
    public void saveBatch(List<Person> personList) {
        mongoTemplate.insertAll(personList);
    }
}
